
#ifndef NMS_FAETUREMAP_TENSOR_H
#define NMS_FAETUREMAP_TENSOR_H

#include <cuda_runtime.h>
#include "../rt_mem_fly.hpp"
#include "cuda_common.hpp"
namespace algocommon
{

// 3*3 kernel 1 stride 1pad
void NMSFeatureMapTensorStream(TensorFloat &source_tensor,
                        TensorFloat &target_tensor,
                        int batch_size,
                        cudaStream_t stream);

}//namespace algocommon

#endif